from langchain_openai import ChatOpenAI
from langchain.memory import SimpleMemory,ConversationBufferMemory
from langchain.prompts import PromptTemplate
from langchain.chains import LLMChain
api_key = "sk-6S0PtpNia71gjcfwSsDPsJ9mGqsVPr2XRQzAx1dHbJS7RW4t"
api_base="https://chatapi.littlewheat.com/v1"

llm = ChatOpenAI(model="gpt-3.5-turbo",api_key=api_key ,base_url=api_base)

template = """以下是人类与AI之间的友好对话描述。AI表现得很健谈，并提供了大量来自其上下文的具体细节。如果AI不知道问题的答案，它会真诚地表示不知道。

当前对话：
{history}
Human: {input}
AI:"""

prompt = PromptTemplate(template=template,input_variable=["history","input"])

chain = LLMChain(llm=llm,
                 prompt=prompt,
                 memory=ConversationBufferMemory(memory_key="history"),
                 verbose=True)

chain.invoke({"input":"你好，请你介绍一下你自己"})
chain.invoke({"input":"我是胡歌，我每天都在学习AI大模型的知识。"})
response = chain.invoke({"input":"请问，胡歌每天都在学习什么？"})
print(response)